
package cup.modules.system.controller;


import cn.hutool.core.map.MapUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.exceptions.ApiException;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import cup.common.common.MyRoleConstant;
import cup.core.log.exception.ServiceException;
import cup.core.redis.cache.CupRedis;
import cup.modules.system.entity.Dept;
import cup.modules.system.entity.Region;
import cup.modules.system.entity.User;
import cup.modules.system.excel.MyUserUploadExcel;
import cup.modules.system.excel.UserExcel;
import cup.modules.system.excel.UserImporter;
import cup.modules.system.service.CqzjLoginService;
import cup.modules.system.service.IDeptService;
import cup.modules.system.service.IRegionService;
import cup.modules.system.vo.UserVO;
import cup.modules.system.wrapper.UserWrapper;
import io.swagger.annotations.*;
import lombok.AllArgsConstructor;
import cup.core.cache.utils.CacheUtil;
import cup.core.excel.util.ExcelUtil;
import cup.core.launch.constant.AppConstant;
import cup.core.mp.support.Condition;
import cup.core.mp.support.Query;
import cup.core.secure.CupUser;
import cup.core.secure.annotation.PreAuth;
import cup.core.secure.utils.AuthUtil;
import cup.core.tenant.annotation.NonDS;
import cup.core.tool.api.R;
import cup.core.tool.constant.CupConstant;
import cup.core.tool.constant.RoleConstant;
import cup.core.tool.utils.DateUtil;
import cup.core.tool.utils.StringPool;
import cup.modules.system.service.IUserService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import static cup.core.cache.constant.CacheConstant.USER_CACHE;

/**
 * 控制器
 *
 * @author
 */
@NonDS
//todo  @ApiIgnore
@RestController
@RequestMapping(AppConstant.APPLICATION_USER_NAME)
@AllArgsConstructor
@Api(value = "用户管理", tags = "用户管理")
public class UserController {

	private final IUserService userService;
	private final IRegionService regionService;
	private final IDeptService iDeptService;
	private final CqzjLoginService cqzjLoginService;
	private final CupRedis cupRedis;

	/**
	 * 查询单条
	 */
	@ApiOperationSupport(order = 1)
	@ApiOperation(value = "查看详情", notes = "传入id")
	@GetMapping("/detail")
//	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R<UserVO> detail(User user) {
		User detail = userService.getOne(Condition.getQueryWrapper(user));
		UserVO userVO = UserWrapper.build().entityVO(detail);
		if (StringUtils.isNotBlank(detail.getDeptId())) {
			Dept dept = iDeptService.getOne(new QueryWrapper<Dept>().eq("id", Long.valueOf(detail.getDeptId())));
			Region region = regionService.getOne(new QueryWrapper<Region>().eq("code", dept.getRegionId()));
			if (null != region) {
				userVO.setRegionCode(region.getCityCode());
				userVO.setDistrictCode(region.getDistrictCode());
			}
		}

		return R.data(userVO);
	}

	/**
	 * 查询单条
	 */
	@ApiOperationSupport(order = 2)
	@ApiOperation(value = "查看详情", notes = "传入id")
	@GetMapping("/info")
	public R<UserVO> info(CupUser user) {
		User detail = userService.getById(user.getUserId());
		return R.data(UserWrapper.build().entityVO(detail));
	}

	/**
	 * 用户列表
	 */
	@GetMapping("/list")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "account", value = "账号名", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "realName", value = "姓名", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "列表", notes = "传入account和realName")
	@PreAuth(MyRoleConstant.HAS_ROLE_CITY_ADMIN_APPROVAL)
	public R<IPage<UserVO>> list(@ApiIgnore @RequestParam Map<String, Object> user, Query query, CupUser bladeUser) {
		QueryWrapper<User> queryWrapper;
		//是否允许在1、2级目录下查询用户 lower=1/deptId=xxx  存在时注入逻辑
		if (user.containsKey("deptId") && user.containsKey("lower") && MapUtil.getInt(user, "lower").equals(1)) {
			String deptId = MapUtil.getStr(user, "deptId");
			user.remove("lower");
			user.remove("deptId");
			queryWrapper = Condition.getQueryWrapper(user, User.class);
			//and(xxx or xxx)
			queryWrapper.and(x -> x.isNull("dept_id").or().in("dept_id", iDeptService.findDeptAndChild(deptId)));
		} else {
			//默认删除一次lower 避免生成sql错误
			user.remove("lower");
			queryWrapper = Condition.getQueryWrapper(user, User.class);
		}
		IPage<User> pages = userService.page(Condition.getPage(query), (!bladeUser.getTenantId().equals(CupConstant.ADMIN_TENANT_ID)) ? queryWrapper.lambda().eq(User::getTenantId, bladeUser.getTenantId()) : queryWrapper);
		return R.data(UserWrapper.build().pageVO(pages));
	}

	/**
	 * 自定义用户列表
	 */
	@GetMapping("/page")
	@ApiImplicitParams({
		@ApiImplicitParam(name = "account", value = "账号名", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "realName", value = "姓名", paramType = "query", dataType = "string")
	})
	@ApiOperationSupport(order = 3)
	@ApiOperation(value = "列表", notes = "传入account和realName")
//	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R<IPage<UserVO>> page(@ApiIgnore User user, Query query, Long deptId, CupUser bladeUser) {
		IPage<User> pages = userService.selectUserPage(Condition.getPage(query), user, deptId, (bladeUser.getTenantId().equals(CupConstant.ADMIN_TENANT_ID) ? StringPool.EMPTY : bladeUser.getTenantId()));
		return R.data(UserWrapper.build().pageVO(pages));
	}

	/**
	 * 新增或修改
	 */
	@PostMapping("/submit")
	@ApiOperationSupport(order = 4)
	@ApiOperation(value = "新增或修改", notes = "传入User")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R submit(@Valid @RequestBody User user) {
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, true) + "*");
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, false) + "*");
		if (null == user.getId()) {
			user.setIsActivity(-1);
		}
		return R.status(userService.submit(user));
	}

	/**
	 * 修改
	 */
	@PostMapping("/update")
	@ApiOperationSupport(order = 5)
	@ApiOperation(value = "修改", notes = "传入User")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R update(@Valid @RequestBody User user) {
//		CacheUtil.clear(USER_CACHE);
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, true) + "*");
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, false) + "*");
		return R.status(userService.updateUser(user));
	}

	/**
	 * 删除
	 */
	@PostMapping("/remove")
	@ApiOperationSupport(order = 6)
	@ApiOperation(value = "删除", notes = "传入id集合")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R remove(@RequestParam String ids) {
//		CacheUtil.clear(USER_CACHE);
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, true) + "*");
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, false) + "*");
		return R.status(userService.removeUser(ids));
	}

	/**
	 * 设置菜单权限
	 */
	@PostMapping("/grant")
	@ApiOperationSupport(order = 7)
	@ApiOperation(value = "权限设置", notes = "传入roleId集合以及menuId集合")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R grant(@ApiParam(value = "userId集合", required = true) @RequestParam String userIds,
				   @ApiParam(value = "roleId集合", required = true) @RequestParam String roleIds) {
		boolean temp = userService.grant(userIds, roleIds);
		return R.status(temp);
	}

	/**
	 * 重置密码
	 */
	@PostMapping("/reset-password")
	@ApiOperationSupport(order = 8)
	@ApiOperation(value = "初始化密码", notes = "传入userId集合")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R resetPassword(@ApiParam(value = "userId集合", required = true) @RequestParam String userIds) {
		String password = userService.resetPassword(userIds);
		return R.data(password);
	}

	/**
	 * 修改密码
	 */
	@PostMapping("/update-password")
	@ApiOperationSupport(order = 9)
	@ApiOperation(value = "修改密码", notes = "传入密码")
	public R updatePassword(CupUser user, @ApiParam(value = "旧密码", required = true) @RequestParam String oldPassword,
							@ApiParam(value = "新密码", required = true) @RequestParam String newPassword,
							@ApiParam(value = "新密码", required = true) @RequestParam String newPassword1) {
		boolean temp = userService.updatePassword(user.getUserId(), oldPassword, newPassword, newPassword1);
		return R.status(temp);
	}

	/**
	 * 修改基本信息
	 */
	@PostMapping("/update-info")
	@ApiOperationSupport(order = 10)
	@ApiOperation(value = "修改基本信息", notes = "传入User")
	public R updateInfo(@Valid @RequestBody User user) {
//		CacheUtil.clear(USER_CACHE);
		//验证用户信息
		Long userId = AuthUtil.getUserId();
		if(!user.getId().equals(userId)){
			throw new ServiceException("用户校验失败,请确认后再提交!");
		}
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, true) + "*");
		cupRedis.del(CacheUtil.formatCacheName(USER_CACHE, false) + "*");
		return R.status(userService.updateUserInfo(user));
	}

	/**
	 * 用户列表
	 */
	@GetMapping("/user-list")
	@ApiOperationSupport(order = 11)
	@ApiOperation(value = "用户列表", notes = "传入user")
	public R<List<User>> userList(User user, CupUser bladeUser) {
		QueryWrapper<User> queryWrapper = Condition.getQueryWrapper(user);
		List<User> list = userService.list((!AuthUtil.isAdministrator()) ? queryWrapper.lambda().eq(User::getTenantId, bladeUser.getTenantId()) : queryWrapper);
		return R.data(list);
	}

	/**
	 * 导入用户
	 */
	@PostMapping("import-user")
	@ApiOperationSupport(order = 12)
	@ApiOperation(value = "导入用户", notes = "传入excel")
	public R importUser(MultipartFile file, Integer isCovered) {
		UserImporter userImporter = new UserImporter(userService, isCovered == 1);
		ExcelUtil.save(file, userImporter, UserExcel.class);
		return R.success("操作成功");
	}

	/**
	 * 导出用户
	 */
	@GetMapping("export-user")
	@ApiOperationSupport(order = 13)
	@ApiOperation(value = "导出用户", notes = "传入user")
	public void exportUser(@ApiIgnore @RequestParam Map<String, Object> user, CupUser bladeUser, HttpServletResponse response) {
		QueryWrapper<User> queryWrapper = Condition.getQueryWrapper(user, User.class);
		if (!AuthUtil.isAdministrator()) {
			queryWrapper.lambda().eq(User::getTenantId, bladeUser.getTenantId());
		}
		queryWrapper.lambda().eq(User::getIsDeleted, CupConstant.DB_NOT_DELETED);
		List<UserExcel> list = new ArrayList<>();
		List<User> users = userService.list(queryWrapper);
		if (!CollectionUtils.isEmpty(users)) {
			for (User u : users) {
				UserExcel userExcel = new UserExcel();
				BeanUtils.copyProperties(u, userExcel);
				list.add(userExcel);
			}
		}
		userService.exportUser(list);
		ExcelUtil.export(response, "用户数据" + DateUtil.time(), "用户数据表", list, UserExcel.class);
	}

	/**
	 * 导出模板
	 */
	@GetMapping("export-template")
	@ApiOperationSupport(order = 14)
	@ApiOperation(value = "导出模板")
	public void exportUser(HttpServletResponse response) {
		List<UserExcel> list = new ArrayList<>();
		ExcelUtil.export(response, "用户数据模板", "用户数据表", list, UserExcel.class);
	}


	/**
	 * 第三方注册用户
	 */
	@PostMapping("/register-guest")
	@ApiOperationSupport(order = 15)
	@ApiOperation(value = "第三方注册用户", notes = "传入user")
	public R registerGuest(User user, Long oauthId) {
		return R.status(userService.registerGuest(user, oauthId));
	}

	/**
	 * 配置用户平台信息
	 */
	@PostMapping("/update-platform")
	@ApiOperationSupport(order = 16)
	@ApiOperation(value = "配置用户平台信息", notes = "传入user")
	public R updatePlatform(Long userId, Integer userType, String userExt) {
		return R.status(userService.updatePlatform(userId, userType, userExt));
	}

	/**
	 * 查看平台详情
	 */
	@ApiOperationSupport(order = 17)
	@ApiOperation(value = "查看平台详情", notes = "传入id")
	@GetMapping("/platform-detail")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	public R<UserVO> platformDetail(User user) {
		return R.data(userService.platformDetail(user));
	}


	/**
	 * 用户列表查询
	 */
	@ApiImplicitParams({
		@ApiImplicitParam(name = "name", value = "人员姓名", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "deptName", value = "部门名称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "postName", value = "职位名称", paramType = "query", dataType = "string"),
		@ApiImplicitParam(name = "current", value = "当前页数", paramType = "query", dataType = "int"),
		@ApiImplicitParam(name = "size", value = "每页数量", paramType = "query", dataType = "int")
	})
	@ApiOperationSupport(order = 18)
	@ApiOperation(value = "用户列表查询", notes = "用户列表查询")
	@GetMapping("/search/user")
	public R<IPage<UserVO>> userSearch(@ApiIgnore UserVO user, @ApiIgnore Query query) {
		return R.data(userService.selectUserSearch(user, query));
	}


	@PostMapping("/uploadUserInfo")
	@PreAuth(RoleConstant.HAS_ROLE_ADMIN)
	@ApiOperation(value = "上传用户信息excel", notes = "上传用户信息excel")
	public R uploadBaseUser(@RequestParam("file") MultipartFile file) {
		List<MyUserUploadExcel> read = ExcelUtil.read(file, MyUserUploadExcel.class);
		if (CollectionUtils.isEmpty(read)) {
			R.success("未查询到数据");
		}
		cqzjLoginService.uploadBaseUser01(read);
		return R.status(true);
	}

	/**
	 * @Description: 根据部门id查询用户
	 * @Author: LuanKay
	 * @Date: 2022/8/9
	 */
	@GetMapping("/findByDeptId")
	@ApiOperation(value = "根据部门id查询用户信息")
	public R<IPage<User>> findByDeptId(User user, Query query) {
		UserWrapper build = UserWrapper.build();
		IPage<User> byDeptId = userService.findByDeptId(Condition.getPage(query), user);
		if (!CollectionUtils.isEmpty(byDeptId.getRecords())) {
			List<User> records = byDeptId.getRecords();
			for (int i = 0; i < records.size(); i++) {
				records.set(i,build.entityVO(records.get(i)));
			}
		}
		return R.data(byDeptId);
	}
}
